#!/bin/bash
#SBATCH -J img_gen_1
#SBATCH -o img_gen_1.out
#SBATCH -p A100_40G
#SBATCH -N 1
#SBATCH --gres=gpu:1
#SBATCH --cpus-per-task=8

python /home/qlc/model/jbhi/img_gen/run.py \
--max_epochs=150 \
--val_every=10 \
--logdir=/home/qlc/train_log/img_gen/t1c \
--modality=t1c \
--gli_encoder=/home/qlc/train_log/pre_gli/t1c/best_gli_encoder.pt \
--text_encoder=/home/qlc/train_log/pre_text/best_text_encoder.pt \
--img_encoder=/home/qlc/train_log/pre_img/t1c/best_mri_encoder.pt 

python /home/qlc/model/jbhi/img_gen/run.py \
--max_epochs=150 \
--val_every=10 \
--logdir=/home/qlc/train_log/img_gen/t2f \
--modality=t2f \
--gli_encoder=/home/qlc/train_log/pre_gli/t2f/best_gli_encoder.pt \
--text_encoder=/home/qlc/train_log/pre_text/best_text_encoder.pt \
--img_encoder=/home/qlc/train_log/pre_img/t2f/best_mri_encoder.pt 


